import requests


class EmpManageApi(object):
    # 添加员工
    def add_emp(self, req_header, req_body):
        resp = requests.post(url="http://ihrm2-test.itheima.net/api/sys/user",
                             headers=req_header, json=req_body)

        return resp

    # 查询员工
    def query_emp(self, emp_id, req_header):
        resp = requests.get(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                            headers=req_header)

        return resp

    # 修改员工
    def modify_emp(self, emp_id, req_header, req_body):
        resp = requests.put(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                            headers=req_header, json=req_body)

        return resp

    # 删除员工
    def del_emp(self, emp_id, req_header):
        resp = requests.delete(url="http://ihrm2-test.itheima.net/api/sys/user/" + emp_id,
                               headers=req_header)

        return resp


if __name__ == '__main__':
    emp = EmpManageApi()
    resp = requests.post(url="http://ihrm2-test.itheima.net/api/sys/login",
                         json={"mobile": "13800000002", "password": "123456"})
    token = resp.json().get("data")
    req_header = {"Content-Type": "application/json",
                  "Authorization": token}
    #  添加员工
    # req_body = {
    #     "username": "小白007",
    #     "mobile": "13193916720",
    #     "workNumber": "9527"
    # }
    # resp = emp.add_emp(req_header,req_body)
    # print("添加员工成功：",resp.json())

    #  查询员工
    emp_id = "1545659477808656384"
    resp = emp.query_emp(emp_id, req_header)
    print("查询员工",resp.json())

